package marker.micro.gateway.filter;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;

import org.springframework.cloud.netflix.zuul.filters.route.ZuulFallbackProvider;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.client.ClientHttpResponse;
import org.springframework.stereotype.Component;

@Component
public class ServiceFallbackProvider implements ZuulFallbackProvider {

    @Override
    public String getRoute() {
	return "*";
    }

    @Override
    public ClientHttpResponse fallbackResponse() {
	return new ClientHttpResponse() {
	    @Override
	    public HttpStatus getStatusCode() throws IOException {
		return HttpStatus.OK;
	    }

	    @Override
	    public int getRawStatusCode() throws IOException {
		return 200;
	    }

	    @Override
	    public String getStatusText() throws IOException {
		return "OK";
	    }

	    @Override
	    public void close() {

	    }

	    @Override
	    public InputStream getBody() throws IOException {
		return new ByteArrayInputStream("服务故障，请稍后重试!".getBytes());
	    }

	    @Override
	    public HttpHeaders getHeaders() {
		HttpHeaders headers = new HttpHeaders();
		headers.setContentType(MediaType.APPLICATION_JSON_UTF8);
		return headers;
	    }
	};
    }

}
